import os

class Config:
    # 配置API密钥
    API_KEYS = {
        "chatgpt": os.getenv('CHATGPT_API_KEY'),
        "deepseek": os.getenv('DEEPSEEK_API_KEY'),
        "other": os.getenv('OTHER_API_KEY')
    }

    # 配置选择的模型
    SELECTED_MODEL = os.getenv('SELECTED_MODEL', 'chatgpt')  # 默认选择 chatgpt

    # 模型的基础 URL 和模型名称
    API_URLS = {
        "chatgpt": "https://api.openai.com/v1/chat/completions",
        "deepseek": "https://api.deepseek.com/v1/chat/completions",
        "other": "https://api.other.com/v1/chat/completions"
    }

    MODEL_NAMES = {
        "chatgpt": "gpt-3.5-turbo",
        "deepseek": "deepseek-model",
        "other": "other-model"
    }

    MAX_TOKENS = 1000
    TEMPERATURE = 0.7
